import os
import json

# 将股票数据保存到JSON文件
def save_to_json(data, filename):
    # 确保输出目录存在
    # __file__ 是当前文件(fileUtils.py)的路径，不是调用这个方法的地方的路径
    # 使用os.path.dirname获取当前文件所在目录，然后向上一级找到files目录
    output_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "files")
    os.makedirs(output_dir, exist_ok=True)

    # 构建完整的文件路径
    file_path = os.path.join(output_dir, filename)

    # 将数据写入JSON文件
    with open(file_path, 'w', encoding='utf-8') as f:
        json.dump(data, f, ensure_ascii=False, indent=2)

    return file_path

def read_from_json(filename):
    # 构建完整的文件路径
    output_dir = os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", "files")
    file_path = os.path.join(output_dir, filename)
    if not os.path.exists(file_path):
        return None
    # 读取JSON文件
    with open(file_path, 'r', encoding='utf-8') as f:
        data = json.load(f)
    return data